from core.domain.core_domain_entities.page import Page
from core.domain.value_object import ValueObject


class PagesClassifier(ValueObject):
    def classify_page(self, page: Page, expected_class):
        """
        This is test example, you need to train classifier model, add onnx to init, predict function
        and some settings to make it actually work
        """
        return expected_class
